package com.lagou.edu.factory;

import com.lagou.edu.annos.Autowired;
import com.lagou.edu.annos.Service;
import com.lagou.edu.annos.Transactional;
import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.Element;
import org.dom4j.Node;
import org.dom4j.io.SAXReader;

import java.io.File;
import java.io.InputStream;
import java.lang.annotation.Annotation;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @author 应癫
 *
 * 工厂类，生产对象（使用反射技术）
 */
public class BeanFactory {

    /**
     * 任务一：读取解析xml，通过反射技术实例化对象并且存储待用（map集合）
     * 任务二：对外提供获取实例对象的接口（根据id获取）
     */

    private static Map<String,Object> map = new HashMap<>();  // 存储对象

    // 存储所有类文件的路径
    private static List<String> cNameList = new ArrayList<>();

    // bean集合
    private static List<Element> beanList = new ArrayList<>();

    static {
        // 任务一：读取解析xml，通过反射技术实例化对象并且存储待用（map集合）
        // 加载xml
        InputStream resourceAsStream = BeanFactory.class.getClassLoader().getResourceAsStream("beans.xml");
        // 解析xml
        SAXReader saxReader = new SAXReader();
        try {
            Document document = saxReader.read(resourceAsStream);
            Element rootElement = document.getRootElement();

            // xml方式：ioc容器实现（bean创建、依赖注入）
            beanList = rootElement.selectNodes("//bean");
            readXmlBeans();

            // 实例化完成之后维护对象的依赖关系，检查哪些对象需要传值进入，根据它的配置，我们传入相应的值
            // 有property子元素的bean就有传值需求
            List<Element> propertyList = rootElement.selectNodes("//property");
            doXmlDI(propertyList);

            // 注解方式：
            // 读取配置类，如果开启了注解，则通过注解方式注册bean
            Element scanEle = (Element) rootElement.selectSingleNode("//component-scan");
            if (null != scanEle) {
                // 开启了注解，读取注解配置
                String packagePath = scanEle.attributeValue("base-package");

                // 扫描包路径下所有类文件，拿到所有文件的路径
                doScan(packagePath);

                // 读取注解@Service，存储到bean
                readServiceBeans();

                // 注解方式的依赖注入
                doAnnotationDI();
            }

            // Transactional注解
            readTransactionalBeans();

        } catch (DocumentException e) {
            e.printStackTrace();
        }

    }

    // 任务二：对外提供获取实例对象的接口（根据id获取）
    public static  Object getBean(String id) {
        return map.get(id);
    }

    public static void doAnnotationDI() {
        for (String key : map.keySet()) {
            Object bean = map.get(key);

            Class clazz = bean.getClass();

            Field[] fields = clazz.getDeclaredFields();

            Method[] methods = clazz.getMethods();

            // 遍历所有属性
            for (Field field : fields) {
                // 是否为自动注入的注解
                if (field.isAnnotationPresent(Autowired.class)) {
                    // 属性的类名称，切割数组
                    String[] nameArr = field.getType().getName().split("\\.");
                    String className = nameArr[nameArr.length - 1];

                    // 遍历所有方法
                    for (Method method : methods) {
                        if(method.getName().equalsIgnoreCase("set" + className)) {
                            // 反射
                            try {
                                method.invoke(bean, map.get(className));
                            } catch (IllegalAccessException e) {
                                e.printStackTrace();
                            } catch (InvocationTargetException e) {
                                e.printStackTrace();
                            }
                        }
                    }

                }
            }

            map.put(key, bean);
        }
    }


    /**
     *
     * @param propertyList
     */
    public static void doXmlDI(List<Element> propertyList) {
        // 解析property，获取父元素
        for (int i = 0; i < propertyList.size(); i++) {
            Element element =  propertyList.get(i);   //<property name="AccountDao" ref="accountDao"></property>
            String name = element.attributeValue("name");
            String ref = element.attributeValue("ref");

            // 找到当前需要被处理依赖关系的bean
            Element parent = element.getParent();

            // 调用父元素对象的反射功能
            String parentId = parent.attributeValue("id");
            Object parentObject = map.get(parentId);
            // 遍历父对象中的所有方法，找到"set" + name
            Method[] methods = parentObject.getClass().getMethods();
            for (int j = 0; j < methods.length; j++) {
                Method method = methods[j];
                if(method.getName().equalsIgnoreCase("set" + name)) {  // 该方法就是 setAccountDao(AccountDao accountDao)

                    try {
                        method.invoke(parentObject,map.get(ref));
                    } catch (IllegalAccessException e) {
                        e.printStackTrace();
                    } catch (InvocationTargetException e) {
                        e.printStackTrace();
                    }

                }
            }

            // 把处理之后的parentObject重新放到map中
            map.put(parentId,parentObject);

        }
    }

    /**
     * 递归扫描路径下的所有文件，获取其类全限定名
     * @param packagePath
     */
    public static void doScan(String packagePath) {
        if (null == packagePath || packagePath.trim().equals("")) {
            throw new RuntimeException("注解包路径base-package不能为空");
        }

        String rootPath = Thread.currentThread().getContextClassLoader().getResource("").getPath();
        rootPath += packagePath.replaceAll("\\.", "/");

        File pkg = new File(rootPath);

        File[] fileList = pkg.listFiles();

        for (File file : fileList) {
            if (file.isDirectory()) {
                doScan(packagePath + '.' + file.getName());
            } else if (file.getName().endsWith(".class")) {
                cNameList.add(packagePath + "." + file.getName().replaceAll(".class", ""));
            }
        }
    }

    public static void readXmlBeans() {
        for (int i = 0; i < beanList.size(); i++) {
            Element element =  beanList.get(i);
            // 处理每个bean元素，获取到该元素的id 和 class 属性
            String id = element.attributeValue("id");        // accountDao
            String clazz = element.attributeValue("class");  // com.lagou.edu.dao.impl.JdbcAccountDaoImpl
            // 通过反射技术实例化对象
            Class<?> aClass = null;
            try {
                aClass = Class.forName(clazz);
                Object o = aClass.newInstance();  // 实例化之后的对象

                // 存储到map中待用
                map.put(id,o);

            } catch (ClassNotFoundException e) {
                e.printStackTrace();
            } catch (InstantiationException e) {
                e.printStackTrace();
            } catch (IllegalAccessException e) {
                e.printStackTrace();
            }
        }
    }

    /**
     * 读取@service注解Bean
     */
    public static void readServiceBeans() {
        for (String filePath : cNameList) {
            try {
                Class clazz = Class.forName(filePath);

                Annotation serviceAnno = clazz.getAnnotation(Service.class);

                if (null != serviceAnno) {
                    Object bean = clazz.newInstance();

                    Service service = (Service) serviceAnno;

                    String id = service.value();
                    if (null != id && !id.trim().equals("")) {
                        map.put(id, bean);
                    } else {
                        String[] nameArr = clazz.getName().split("\\.");
                        String name = nameArr[nameArr.length - 1];

                        map.put(name, bean);
                    }
                }

            } catch (ClassNotFoundException e) {
                e.printStackTrace();
            } catch (InstantiationException e) {
                e.printStackTrace();
            } catch (IllegalAccessException e) {
                e.printStackTrace();
            }

        }
    }


    public static void readXml() {
        // 任务一：读取解析xml，通过反射技术实例化对象并且存储待用（map集合）
        // 加载xml
        InputStream resourceAsStream = BeanFactory.class.getClassLoader().getResourceAsStream("beans.xml");
        // 解析xml
        SAXReader saxReader = new SAXReader();
        try {
            Document document = saxReader.read(resourceAsStream);
            Element rootElement = document.getRootElement();
            List<Element> beanList = rootElement.selectNodes("//bean");
            for (int i = 0; i < beanList.size(); i++) {
                Element element =  beanList.get(i);
                // 处理每个bean元素，获取到该元素的id 和 class 属性
                String id = element.attributeValue("id");        // accountDao
                String clazz = element.attributeValue("class");  // com.lagou.edu.dao.impl.JdbcAccountDaoImpl
                // 通过反射技术实例化对象
                Class<?> aClass = Class.forName(clazz);
                Object o = aClass.newInstance();  // 实例化之后的对象

                // 存储到map中待用
                map.put(id,o);

            }

            // 实例化完成之后维护对象的依赖关系，检查哪些对象需要传值进入，根据它的配置，我们传入相应的值
            // 有property子元素的bean就有传值需求
            List<Element> propertyList = rootElement.selectNodes("//property");
            // 解析property，获取父元素
            for (int i = 0; i < propertyList.size(); i++) {
                Element element =  propertyList.get(i);   //<property name="AccountDao" ref="accountDao"></property>
                String name = element.attributeValue("name");
                String ref = element.attributeValue("ref");

                // 找到当前需要被处理依赖关系的bean
                Element parent = element.getParent();

                // 调用父元素对象的反射功能
                String parentId = parent.attributeValue("id");
                Object parentObject = map.get(parentId);
                // 遍历父对象中的所有方法，找到"set" + name
                Method[] methods = parentObject.getClass().getMethods();
                for (int j = 0; j < methods.length; j++) {
                    Method method = methods[j];
                    if(method.getName().equalsIgnoreCase("set" + name)) {  // 该方法就是 setAccountDao(AccountDao accountDao)
                        method.invoke(parentObject,map.get(ref));
                    }
                }

                // 把处理之后的parentObject重新放到map中
                map.put(parentId,parentObject);

            }


        } catch (DocumentException e) {
            e.printStackTrace();
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        } catch (InstantiationException e) {
            e.printStackTrace();
        } catch (InvocationTargetException e) {
            e.printStackTrace();
        }
    }


    /**
     *
     */
    public static void readTransactionalBeans(){
        for (String key : map.keySet()) {
            Object bean = map.get(key);

            Class clazz = bean.getClass();

            if(clazz.isAnnotationPresent(Transactional.class)) {
                //获取代理工厂
                ProxyFactory proxyFactory = (ProxyFactory) BeanFactory.getBean("ProxyFactory");

                Class[] interfaces = clazz.getInterfaces();

                if (null != interfaces && interfaces.length > 0) {
                    bean = proxyFactory.getJdkProxy(bean);
                } else {
                    bean = proxyFactory.getCglibProxy(bean);
                }

            }

            map.put(key, bean);
        }
    }
}
